package core

import (
	"fmt"
	"github.com/spf13/viper"
)

func Viper() *viper.Viper {
	// 后续可改为根据环境来获取对应的 config 配置
	config := "config.yaml"

	v := viper.New()
	v.SetConfigFile(config)
	v.SetConfigType("yaml")
	err := v.ReadInConfig()

	if err != nil {
		panic(fmt.Errorf("fatal error config file: %s", err))
	}
	if err = v.Unmarshal(config); err != nil {
		fmt.Println(err)
	}

	return v
}
